from flask import jsonify

class Response:

    def data(self, data):
        if(type(data) == type('')):
            return data
        return jsonify(data)
    
    def success(self, message,data=None):
        return jsonify({
            'message': message,
            'status': True,
            'code': 201,
            'data': data
        })
    
    def error(self, message, code=400):
        return jsonify({
            'message': message,
            'status': False,
            'code': code,
            'data': None
        })